Fix selenium tests