fix module injection for ml.utils