Update tests to test Go button