changed logic of test