Fix minor error in tests that made it fail