Removed unnecessary import from base.py