Fixed another import

Yet a damage from the latest merge