Changed one statement to fix unit test.