Fix imported things needed for regulator.py to work