Fix wrong count tests

Else, the second condition is wrong and will never be run.