fix module imports in example