Fix message appearing even after you run code