fixed import dependency in cleanup.py