Correct the import on readme example