Fixing selector on broken test