Fixed the issue where the ending freezes for multiple seconds (again cause someone broke my previous fix)