Fix demo import statement