fix #3 wrong import problem