Fold: Bug fix

Fixed bug that prevented the end screen from being shown.