diff --git a/src/utils/get_surprise.jl b/src/utils/get_surprise.jl
index acdcf25..24b1667 100644
--- a/src/utils/get_surprise.jl
+++ b/src/utils/get_surprise.jl
@@ -61,6 +61,12 @@ Equation:
 """
 function get_surprise(node::ContinuousInputNode)
 
+    #If there was no input
+    if ismissing(node.states.input_value)
+        #Return no surprise
+        return 0
+    end
+
     #Sum the predictions of the vaue parents
     parents_prediction_mean = 0
     for parent in node.edges.observation_parents
@@ -85,6 +91,12 @@ Calculate the surprise of a binary input node on seeing the last input.
 
 function get_surprise(node::BinaryInputNode)
 
+    #If there was no input
+    if ismissing(node.states.input_value)
+        #Return no surprise
+        return 0
+    end
+
     #Sum the predictions of the vaue parents
     parents_prediction_mean = 0
     for parent in node.edges.observation_parents
@@ -137,6 +149,12 @@ Calculate the surprise of a categorical input node on seeing the last input.
 """
 function get_surprise(node::CategoricalInputNode)
 
+    #If there was no input
+    if ismissing(node.states.input_value)
+        #Return no surprise
+        return 0
+    end
+
     #Get value parent
     parent = node.edges.observation_parents[1]