Slight modifications to session test