Fix condition from drop collection unit test