fix imports in example