diff --git a/sohstationviewer/view/plotting/plotting_widget/plotting_axes.py b/sohstationviewer/view/plotting/plotting_widget/plotting_axes.py
index caeff9197fce4f13bcf63e8bd709e3dd24059633..f2eb555f9c70fb5354479ca9dcf4404f7b783e60 100644
--- a/sohstationviewer/view/plotting/plotting_widget/plotting_axes.py
+++ b/sohstationviewer/view/plotting/plotting_widget/plotting_axes.py
@@ -2,6 +2,7 @@ from typing import List, Optional, Dict
 
 import numpy as np
 from matplotlib.axes import Axes
+from matplotlib.text import Text
 from matplotlib.patches import ConnectionPatch, Rectangle
 from matplotlib.ticker import AutoMinorLocator
 from matplotlib import pyplot as pl
@@ -154,7 +155,7 @@ class PlottingAxes:
         return ax
 
     def create_sample_no_label(self, ax: Axes, pos_y: float,
-                               sample_no: int, color: str) -> None:
+                               sample_no: int, color: str) -> Optional[Text]:
         """
         Create label to display sample number
         :param ax: the axes of the label