diff --git a/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py b/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py
index f025dad0..2bcffe9c 100644
--- a/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py
+++ b/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py
@@ -49,7 +49,7 @@ def get_out_going_vertices(
 
     @overrides(AbstractSplitterCommon.get_in_coming_vertices)
     def get_in_coming_vertices(
-            self, partition_id: str) -> List[SourceSegmentedSDRAMMachinePartition]:
+            self, partition_id: str) -> List[SDRAMMachineVertex]:
         return self._pre_vertices