Add back forgotten import.
0.7.1
Merge branch 'release/0.7.1' into develop