Add fix to UI tests

Added await after login

Change-Id: Id023d3d42f72b5b32bd73151292f6b1c69eab36f
Closes-Bug: #1287091
Merge "Add fix to UI tests"