Fix import statement in example