Fix mistake in examples

Add missing import