fix selector

Github seems to have updated the class