Fix Assess Student flow after design change.