diff --git a/src/robot_interface/main.py b/src/robot_interface/main.py
index 3272ef4..5ab5266 100644
--- a/src/robot_interface/main.py
+++ b/src/robot_interface/main.py
@@ -65,7 +65,7 @@ def start_video_rcv(robot_ip, port, socket):
     :type port: int
 
     :param socket: The ZMQ socket to send the video images over.
-    :type: Socket
+    :type: VideoSender
     """
     socket.socket.setsockopt(zmq.CONFLATE,1)
     
@@ -94,7 +94,7 @@ def video_rcv_loop(vid_service, vid_stream_name, socket):
     :type vid_service: Object (Qi service object)
 
     :param socket: The ZMQ socket to send the video images over.
-    :type: Socket
+    :type: VideoSender
     """
     while not state.exit_event.is_set():
         try: