Fixing imports in the example