Fix a missing import

This fixes a failure encountered when using the "multirun" feature.