diff --git a/Basis_Function.py b/Basis_Function.py
index 832355fbba93d6ccdd9d5e57886ecf6af4af310a..e434fce15dfc0675e281f3c8b47edd072515c5a0 100644
--- a/Basis_Function.py
+++ b/Basis_Function.py
@@ -13,7 +13,7 @@ x = Symbol('x')
 z = Symbol('z')
 
 
-class Vector:
+class Basis:
     """Class for basis vector.
 
     Attributes
@@ -117,7 +117,7 @@ class Vector:
         pass
 
 
-class Legendre(Vector):
+class Legendre(Basis):
     """Class for Legendre basis."""
     def _build_basis_vector(self, eval_point):
         """Constructs basis vector.
diff --git a/Plotting.py b/Plotting.py
index 3578cbce0dc3a15fec88149a1d0cd7c402fecdac..dd6a58205ad9bcb26323061956145d7f143ee658 100644
--- a/Plotting.py
+++ b/Plotting.py
@@ -18,7 +18,7 @@ from sympy import Symbol
 
 from Quadrature import Quadrature
 from Initial_Condition import InitialCondition
-from Basis_Function import Vector
+from Basis_Function import Basis
 from projection_utils import calculate_exact_solution,\
     calculate_approximate_solution
 from encoding_utils import decode_ndarray
@@ -322,7 +322,7 @@ def plot_evaluation_results(evaluation_file: str, directory: str,
 
 
 def plot_approximation_results(data_file: str, directory: str, plot_name: str,
-                               basis: Vector, quadrature: Quadrature,
+                               basis: Basis, quadrature: Quadrature,
                                init_cond: InitialCondition) -> None:
     """Plots given approximation results.
 
@@ -376,7 +376,7 @@ def plot_approximation_results(data_file: str, directory: str, plot_name: str,
 def plot_results(projection: ndarray, troubled_cell_history: list,
                  time_history: list, mesh: ndarray, num_grid_cells: int,
                  polynomial_degree: int, wave_speed: float, final_time: float,
-                 left_bound: float, right_bound: float, basis: Vector,
+                 left_bound: float, right_bound: float, basis: Basis,
                  quadrature: Quadrature, init_cond: InitialCondition,
                  colors: dict = None, coarse_projection: ndarray = None,
                  multiwavelet_coeffs: ndarray = None) -> None: