From bc3d5357dc1c065e91a4792ed099555fe1ccc1f6 Mon Sep 17 00:00:00 2001
From: Stanislaw Malinowski <stanislaw.malinowski@diamond.ac.uk>
Date: Tue, 26 Nov 2024 17:08:46 +0000
Subject: [PATCH] make all connect

---
 src/dodal/beamlines/i18.py | 18 +++---------------
 1 file changed, 3 insertions(+), 15 deletions(-)

diff --git a/src/dodal/beamlines/i18.py b/src/dodal/beamlines/i18.py
index 5260e746fe..e5cf5e8940 100644
--- a/src/dodal/beamlines/i18.py
+++ b/src/dodal/beamlines/i18.py
@@ -237,10 +237,10 @@ def main_table(
 
 def thor_labs_table(
     wait_for_connection: bool = True, fake_with_ophyd_sim: bool = False
-) -> Table:
+) -> ThorLabsStage:
     return device_instantiation(
-        Table,
-        "table",
+        ThorLabsStage,
+        "thor_labs_stage",
         "-MO-TABLE-02:",
         wait_for_connection,
         fake_with_ophyd_sim,
@@ -262,15 +262,3 @@ def raster_stage(
         wait_for_connection,
         fake_with_ophyd_sim,
     )
-
-
-def thor_labs_table(
-    wait_for_connection: bool = True, fake_with_ophyd_sim: bool = False
-) -> ThorLabsStage:
-    return device_instantiation(
-        ThorLabsStage,
-        "table",
-        "-MO-TABLE-02:",
-        wait_for_connection,
-        fake_with_ophyd_sim,
-    )