1 files changed, 5 insertions, 2 deletions
diff --git a/src/solvers/gpu/host.rs b/src/solvers/gpu/host.rs
index e354263..d67138e 100644
--- a/src/solvers/gpu/host.rs
+++ b/src/solvers/gpu/host.rs
@@ -123,8 +123,11 @@ impl Host {
         println!("finished gpu setup");
         loop {
             match self.receiver.recv().expect("Channel to Host broke") {
+                Message::CpuDone => {
+                    self.output_sender.send(Message::CpuDone);
+                    return;
+                }
                 Message::Terminate => {
-                    self.output_sender.send(Message::Terminate);
                     return;
                 }
                 Message::HostMessage((id, i, buffer)) => {
@@ -184,7 +187,7 @@ impl Host {
                         )))
                         .unwrap();
                 }
-                _ => println!("Invalid MessageType"),
+                m => println!("Invalid MessageType {:?} recived by host", m),
             }
         }
     }