Fixed a bug that was causing issues with the restart button.