2 files changed, 4 insertions, 2 deletions
diff --git a/test/test-client-bandwidth.c b/test/test-client-bandwidth.c
index cb05473..7e3cf8e 100644
--- a/test/test-client-bandwidth.c
+++ b/test/test-client-bandwidth.c
@@ -41,8 +41,9 @@ while (1) {
     printf ("Throughput: %.2fGbyte/s\n",((size * 500) / elapsed)/(1024*1024*1024));
 }
     g_timer_stop (timer);
-    kiro_client_free (client);
+    kiro_trb_purge (trb, FALSE);
     kiro_trb_free (trb);
+    kiro_client_free (client);
     return 0;
 }
 
diff --git a/test/test-client.c b/test/test-client.c
index 4cfc842..7240ab1 100644
--- a/test/test-client.c
+++ b/test/test-client.c
@@ -89,8 +89,9 @@ main ( int argc, char *argv[] )
         render (data_sf);
     }
 
-    kiro_client_free (client);
+    kiro_trb_purge (trb, FALSE);
     kiro_trb_free (trb);
+    kiro_client_free (client);
     return 0;
 }