fixed import dependency in cleanup.py

Perhaps cleanup.py isn't actually used, but this makes it work rather than throwing an error when it gets to line 15.