minor fix: import