diff -r c78f677d96b8 -r 6d9713bfc346 test-driver.c
--- a/test-driver.c	Tue Apr 08 23:53:57 2008 -0400
+++ b/test-driver.c	Sun Jun 15 10:50:16 2008 -0400
@@ -262,8 +262,12 @@
 		razor_transaction_install_package(ctx->trans, pkg);
 	}		
 	for (i = 0; i < ctx->n_remove_pkgs; i++) {
-		pkg = razor_set_get_package(ctx->repo_set,
+		pkg = razor_set_get_package(ctx->system_set,
 					    ctx->remove_pkgs[i]);
+		if (!pkg)
+			pkg = razor_set_get_package(ctx->repo_set,
+						    ctx->remove_pkgs[i]);
+
 		razor_transaction_remove_package(ctx->trans, pkg);
 	}		
 
@@ -278,6 +282,7 @@
 	if (!errors) {
 		struct razor_set *new;
 		new = razor_transaction_finish(ctx->trans);
+		ctx->trans = NULL;
 		ctx->system_set = new;
 	}
 }