diff --git a/CI_files/default_test.m b/CI_files/default_test.m
index ede005b2437db6ff8435939ed22118d9bacc24c6..17b41f5f0918586e6259e484b1753367ddf1bbb9 100644
--- a/CI_files/default_test.m
+++ b/CI_files/default_test.m
@@ -10,64 +10,69 @@ try
     rmdir('CI_files/export','s');
 end
 
-ProjectID = 'Test01';
-%% Data
-% some random data
-x = linspace(0,7);
-y = rand(1,100)+2;
-dataset1 = 'test_data.mat';
-pwd
-save('CI_files/test_data.mat','x','y');
-% some data as .h5
-x1 = linspace(0,2*pi);
-y1 = sin(x1)+2;
-
-% define file path & name
-fpath = "CI_files/testdata2.h5";
-dataset2 = 'CI_files/testdata2.h5';
-
-% create hdf5 file and dataset > write data to hdf5 file / dataset
-h5create(fpath, "/x1", size(x1), "Datatype", class(x1))
-h5create(fpath, "/y1", size(y1), "Datatype", class(y1))
-h5write(fpath, "/x1", x1)
-h5write(fpath, "/y1", y1)
-
-%% Plotting
-
-fig(1) =figure('visible','off');
-plot(x,y,'-k');
-hold on
-plot(x1,y1,'-r');
-
-%% Tag the plot
 try
-    [figs, ID] = PlotID.TagPlot(fig,'ProjectID', ProjectID);
-
-    %% call a dummy function
-    a=1;
-    a = example_fcn(a);
-
-    %% publishing
-
-    % The functions needs the file location, the location of the data and the
-    % figure
-    script = [mfilename('fullpath'),'.m']; % filename of the m.script
-
-    % file name of the data
-    rdata =  {dataset1,dataset2} ; % don't forget the extension
-
-    PlotID.Publish(rdata,script, figs, 'Location', 'CI-Test')
-    result = true;
-    
-    % clean up 
-    delete CI_files/export/* CI_files/*.mat CI_files/*.h5 
-    rmdir('CI_files/export','s'); 
-    
-    clc;
-     
-catch
-    result = false;
-    warning('simple_test failed');
+    ProjectID = 'Test01';
+    %% Data
+    % some random data
+    x = linspace(0,7);
+    y = rand(1,100)+2;
+    dataset1 = 'test_data.mat';
+    pwd
+    save('CI_files/test_data.mat','x','y');
+    % some data as .h5
+    x1 = linspace(0,2*pi);
+    y1 = sin(x1)+2;
+
+    % define file path & name
+    fpath = "CI_files/testdata2.h5";
+    dataset2 = 'CI_files/testdata2.h5';
+
+    % create hdf5 file and dataset > write data to hdf5 file / dataset
+    h5create(fpath, "/x1", size(x1), "Datatype", class(x1))
+    h5create(fpath, "/y1", size(y1), "Datatype", class(y1))
+    h5write(fpath, "/x1", x1)
+    h5write(fpath, "/y1", y1)
+
+    %% Plotting
+
+    fig(1) =figure('visible','off');
+    plot(x,y,'-k');
+    hold on
+    plot(x1,y1,'-r');
+
+    %% Tag the plot
+    try
+        [figs, ID] = PlotID.TagPlot(fig,'ProjectID', ProjectID);
+
+        %% call a dummy function
+        a=1;
+        a = example_fcn(a);
+
+        %% publishing
+
+        % The functions needs the file location, the location of the data and the
+        % figure
+        script = [mfilename('fullpath'),'.m']; % filename of the m.script
+
+        % file name of the data
+        rdata =  {dataset1,dataset2} ; % don't forget the extension
+
+        PlotID.Publish(rdata,script, figs, 'Location', 'CI-Test')
+        result = true;
+
+        % clean up 
+        delete CI_files/export/* CI_files/*.mat CI_files/*.h5 
+        rmdir('CI_files/export','s'); 
+
+        clc;
+
+    catch
+        result = false;
+        warning('simple_test failed in Stage 2');
+    end
+catch    
+     result = false;
+     warning('simple_test failed in Stage 1'); 
 end
 
 end