Fix util import so reloading the module works
Why make this an optional argument...