minor
Added import to fix tutorial example