Fix deprecated import