Updated selenium test to select correct address