Fexed remove button error