minor fix fo import