Fix Deco-grid transform to actually return the classes