diff --git a/common.py b/common.py
index d5c787c..422cfc9 100644
--- a/common.py
+++ b/common.py
@@ -3,7 +3,6 @@ from typing import Dict, List, Tuple
 import matplotlib.pyplot as plt
 import numpy as np
 import numpy.typing as npt
-from numpy._typing import NDArray
 
 # Data given by the assignment
 city_coords: Dict = {
@@ -73,7 +72,7 @@ def plot_plan(city_order: List[str]) -> None:
     plt.show()
 
 
-def index_to_city(indexes: npt.NDArray, cities: npt.NDArray) -> npt.NDArray:
+def indexes_to_cities(indexes: npt.NDArray, cities: npt.NDArray) -> npt.NDArray:
     """Create an array of cities from indeces in a specific order.
 
     Args: