Removing entity from collection fix