Updating imports for the session usage