diff --git a/sharplhf00l04.c b/sharplhf00l04.c
index a4d086e..1caedd0 100644
--- a/sharplhf00l04.c
+++ b/sharplhf00l04.c
@@ -48,7 +48,6 @@
 int probe_lhf00l04(struct flashchip *flash)
 {
 	volatile uint8_t *bios = flash->virtual_memory;
-	volatile uint8_t *registers;
 	uint8_t id1, id2;
 
 #if 0
@@ -75,24 +74,12 @@
 
 	printf_debug("%s: id1 0x%x, id2 0x%x\n", __FUNCTION__, id1, id2);
 
-	if (id1 == flash->manufacture_id && id2 == flash->model_id) {
-		size_t size = flash->total_size * 1024;
-		// we need to mmap the write-protect space. 
-		registers = mmap(0, size, PROT_WRITE | PROT_READ, MAP_SHARED,
-			    fd_mem, (off_t) (0 - 0x400000 - size));
-		if (registers == MAP_FAILED) {
-			// it's this part but we can't map it ...
-			perror("Can't mmap memory using " MEM_DEV);
-			exit(1);
-		}
+	if (id1 != flash->manufacture_id || id2 != flash->model_id) 
+		return 0;
 
-		flash->virtual_registers = registers;
-		printf("bios %p, *bios 0x%x, bios[1] 0x%x\n", bios, *bios,
-		       bios[1]);
-		return 1;
-	}
+	map_flash_registers(flash);
 
-	return 0;
+	return 1;
 }
 
 uint8_t wait_lhf00l04(volatile uint8_t *bios)