Bugfix

Well, that was quick. Fixed it not working on restart.