Fixed infer_shape method.