Changed literally 1 line and now levels restart on death. Woop!