fixing test_shape