Add a missing import to tutorial example