Fix missing import in generic runner