1.1 --- a/test-driver.c	Wed Jun 04 21:28:26 2008 -0400
     1.2 +++ b/test-driver.c	Sun Jun 08 11:56:11 2008 -0400
     1.3 @@ -262,8 +262,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 @@ -278,6 +282,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  }