minor fix in import