Added missing import in megatest