Fix import for recent lazyflow change.