Fix little issue with active class