BUG: fix version module import handling