After login fix