Fix the remove from collection buttton