Fixes broken restart feature