fix input module
add missing imports