Fix regression in ez_setup.py

due to import error