It's getting a little bit more complicated in the @ml_class