Bug/fix selenium tests