clean up restock button logic.