Fix drop collection bug in tests