[#331] Add missing import