WL-1704 Fix for missing import