Slight modification to the sessions module