diff --git a/gui/robot.py b/gui/robot.py
index aea9db9..de9a8dd 100644
--- a/gui/robot.py
+++ b/gui/robot.py
@@ -40,10 +40,15 @@ class ChangeRobot(QWidget):
         self.updateConnectButtonText()
 
     def updateConnectButtonText(self):
+        radioState = True
         if self.status() == "not_connected":
             self.connectButton.setText("Соединить")
         elif self.status() == "connected":
             self.connectButton.setText("Отключить")
+            radioState = False
+            
+        for radioButton in self.robotsRadio:
+            radioButton.setEnabled(radioState)
 
     def timerCallback(self):
         new_status = self.status()
diff --git a/main.py b/main.py
index 654687f..acecd76 100644
--- a/main.py
+++ b/main.py
@@ -2,7 +2,8 @@ import json
 import sys
 import time
 from PyQt5.QtWidgets import QApplication
-from client_socket import SocketRobotArm
+
+from robot.client_socket import SocketRobotArm
 from gui.init import MainContentComponent
 
 class MyApp:
diff --git a/robot/client_socket.py b/robot/client_socket.py
index 951e9c7..25f7809 100644
--- a/robot/client_socket.py
+++ b/robot/client_socket.py
@@ -10,7 +10,7 @@ import pybullet_data
 
 os.environ["LIBGL_ALWAYS_SOFTWARE"] = "1"
 
-from func import *
+from robot.func import *
 
 
 class SocketRobotArm: