[#858] Add import to test