Fix imports in example