Fix css class in js