Fix firing not stopping until you press another key