Missing import for class

When defining a category, the base class must be imported