Fixed mistake on test case