1.1 --- a/src/test-driver.c	Mon Jun 16 15:40:30 2008 -0400
     1.2 +++ b/src/test-driver.c	Mon Jun 23 20:48:48 2008 -0400
     1.3 @@ -264,8 +264,12 @@
     1.4  		razor_transaction_install_package(ctx->trans, pkg);
     1.5  	}
     1.6  	for (i = 0; i < ctx->n_remove_pkgs; i++) {
     1.7 -		pkg = razor_set_get_package(ctx->repo_set,
     1.8 +		pkg = razor_set_get_package(ctx->system_set,
     1.9  					    ctx->remove_pkgs[i]);
    1.10 +		if (!pkg)
    1.11 +			pkg = razor_set_get_package(ctx->repo_set,
    1.12 +						    ctx->remove_pkgs[i]);
    1.13 +
    1.14  		razor_transaction_remove_package(ctx->trans, pkg);
    1.15  	}
    1.16  
    1.17 @@ -280,6 +284,7 @@
    1.18  	if (!errors) {
    1.19  		struct razor_set *new;
    1.20  		new = razor_transaction_finish(ctx->trans);
    1.21 +		ctx->trans = NULL;
    1.22  		ctx->system_set = new;
    1.23  	}
    1.24  }