Space only restarts when the game is over

(I hope)