1 files changed, 7 insertions, 2 deletions
diff --git a/Software/Visual_Studio/Utilities/Tango.Stubs.UI/Resources/CodeTabTemplate.cs b/Software/Visual_Studio/Utilities/Tango.Stubs.UI/Resources/CodeTabTemplate.cs
index 68b249a8a..2a721184b 100644
--- a/Software/Visual_Studio/Utilities/Tango.Stubs.UI/Resources/CodeTabTemplate.cs
+++ b/Software/Visual_Studio/Utilities/Tango.Stubs.UI/Resources/CodeTabTemplate.cs
@@ -7,14 +7,19 @@ using System.Windows.Forms;
 using System.Threading;
 using System.Threading.Tasks;
 using System.Collections.Generic;
+using Tango.PMR.Stubs;
 using Tango.Stubs.UI;
 
 public void OnExecute(StubManager stubManager)
 {
     for (int i = 0; i < 10; i++)
     {
-        stubManager.Run("calculate", 10, 5);
-        Thread.Sleep(10);
+        var response = stubManager.Run<CalculateResponse>("calculate", 10, 5);
+        
+        if (response.Sum == 15)
+        {
+            stubManager.WriteLine("OK!");
+        }
     }
 }