Clarified of if clause, adjusted test case