Fixing selenium class