minor fix import