Updated the assertion to use the abstracted selenium method