Update session flash unit tests