Fixed import in example_test