Fixed import of controller