quick class fix