Fix the delete work bug