Update import to match tutorial docs