Fixes failing test

Updates test to handled updated return to results button.