diff --git a/workflows/go_workflow.py b/workflows/go_workflow.py
index 571f1f8..545a929 100644
--- a/workflows/go_workflow.py
+++ b/workflows/go_workflow.py
@@ -87,8 +87,8 @@ class GODistanceFunction(Function):
 
         return gene_distances
 
-class TestDataFunction(Function):
-    
+
+class TestDataFunction(Function):  
     def __init__(self):
         Function.__init__(self, 'test_data', 'Generate Test Data')
 
@@ -98,7 +98,7 @@ class TestDataFunction(Function):
         axis_0 = ['rows',[]]
         axis_1 = ['cols',[]]
         X = dataset.Dataset(x,[axis_0,axis_1])
-        return [X]
+        return [X, plots.SinePlot(None)]
 
 
 class DatasetInfoFunction(Function):