minor fix
remove unused imports