diff --git a/GPU_version/vio/include/vio/cl_class.h b/GPU_version/vio/include/vio/cl_class.h
index 1b55742..8f1cd4c 100644
--- a/GPU_version/vio/include/vio/cl_class.h
+++ b/GPU_version/vio/include/vio/cl_class.h
@@ -174,7 +174,8 @@ public:
             err=queue->enqueueNDRangeKernel(*_kernels.at(id1)._kernel, cl::NullRange/*offset*/, cl::NDRange(x)/*Global*/, cl::NullRange/*local*/,NULL,&event);
         };
         assert(err==CL_SUCCESS);
-        return event.wait();
+        event.wait();
+        return queue->finish();
 
     }
     template<typename T>