Apply class to correct role