Fixing issue with checkbox field class