CB-5756: Add missing import