Fix the issue of showing assess screen when saving an assessment