SOS-2462 Add missing class in import