Fix missing import
Merge branch 'release/1.0.1'