WFU-3470 Adding missing import