diff --git a/src/Test/IOTasks/Testing.hs b/src/Test/IOTasks/Testing.hs
index 8b60995..78e9345 100644
--- a/src/Test/IOTasks/Testing.hs
+++ b/src/Test/IOTasks/Testing.hs
@@ -24,7 +24,6 @@ import Test.IOTasks.Internal.Output
 import Control.Concurrent.STM
 import Control.Monad (when, forM, replicateM)
 
-import Data.Maybe (catMaybes)
 import Data.List (sortOn)
 import Data.Functor (void)
 import Data.Bifunctor (first)
diff --git a/src/Test/IOTasks/Z3.hs b/src/Test/IOTasks/Z3.hs
index 0b93723..a549b53 100644
--- a/src/Test/IOTasks/Z3.hs
+++ b/src/Test/IOTasks/Z3.hs
@@ -24,7 +24,6 @@ import Test.QuickCheck (generate)
 
 import Control.Concurrent.STM
 
-import Control.Monad (void)
 import Control.Monad.State
 
 import Data.Maybe (catMaybes, fromMaybe, mapMaybe, fromJust)
@@ -69,6 +68,7 @@ data SearchContext = NoContext | LastNotSAT Int | RequirePruningCheck
 -- depth is currently not used to trigger pruning
 updateContext :: SatResult a -> Int -> SearchContext -> SearchContext
 updateContext (SAT _) _ _ = NoContext
+updateContext Timeout _ _ = NoContext
 updateContext NotSAT d NoContext = LastNotSAT d
 updateContext NotSAT _ (LastNotSAT _) = RequirePruningCheck
 updateContext _ _ RequirePruningCheck = error "updateContext: should not happen"
@@ -104,7 +104,10 @@ satPathsQ nVar to t maxUnfolds maxSeqLength checkOverflows q = do
           res <- lift $ isSatPath to s maxSeqLength checkOverflows
           case res of
             NotSAT -> pure ()
-            SAT _ -> do
+            Timeout -> do
+              put NoContext
+              satPaths' nUnfolds nInputs to r (s,d+1) q
+            SAT () -> do
               put NoContext
               satPaths' nUnfolds nInputs to r (s,d+1) q
         _ -> do