fix logic of test two