Fixed an error that lead to test failure