diff --git a/drivers/pci/vgaarb.c b/drivers/pci/vgaarb.c
index 5e6b1eb54c6432b988b86b1f7b30bdd965c17bfb..ed37fd3dcdefc73c73122bb856e4f2a44114fbea 100644
--- a/drivers/pci/vgaarb.c
+++ b/drivers/pci/vgaarb.c
@@ -1261,13 +1261,13 @@ static ssize_t vga_arb_write(struct file *file, const char __user *buf,
 			goto done;
 		}
 
-		vga_put(pdev, io_state);
-
 		if (io_state & VGA_RSRC_LEGACY_IO)
 			uc->io_cnt--;
 		if (io_state & VGA_RSRC_LEGACY_MEM)
 			uc->mem_cnt--;
 
+		vga_put(pdev, io_state);
+
 		ret_val = count;
 		goto done;
 	} else if (strncmp(curr_pos, "trylock ", 8) == 0) {