Fix imports in host