SW-66. Fix missing import