Fix selenium test bug