fix python3 import error
remove unneded imports