Add missing import

Ticket: #6980