fix import of class that was removed
fix import of class that was removed