fix imports in test and example