Unfortunately, the controls are not working when restarting the level. Good idea that could lead to interesting puzzles though!