Change to assert_allclose so I can see the errors