2 changes to stop the errors

add the wait to prevent the selenium cannot find element failure after
the click.
change SHOULD to EXPECT to remove the depreciation warning