Fixed bug where wait box appears when it shouldnae