diff --git a/SlamController/SlamController_pb2_grpc.py b/SlamController/SlamController_pb2_grpc.py
index 16c5743..dd7bdf5 100644
--- a/SlamController/SlamController_pb2_grpc.py
+++ b/SlamController/SlamController_pb2_grpc.py
@@ -1,7 +1,7 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 import grpc
 
-import SlamController_pb2 as SlamController__pb2
+from . import SlamController_pb2 as SlamController__pb2
 
 
 class SlamControlStub(object):
diff --git a/SlamController/slam_streamer.py b/SlamController/slam_streamer.py
index f79366e..f062a36 100644
--- a/SlamController/slam_streamer.py
+++ b/SlamController/slam_streamer.py
@@ -14,11 +14,11 @@ class SlamStreamer:
     can_scan = False
 
     def __init__(self, map_pixels = None, map_meters = None, lidar_connection = None, port = None):
-        self._socket = self._start_socket(self._create_tcp_pub_socket(), port)
         self._map_pixels = map_pixels
         self._map_meters = map_meters
         self._lidar_connection = lidar_connection
         self._port = port
+        self._socket = self._start_socket(self._create_tcp_pub_socket(), port)
 
     def start(self):
         '''