Fix removing models from collections