diff --git a/dkms/post_remove.sh b/dkms/post_remove.sh
index 6cdfece..0b735ed 100755
--- a/dkms/post_remove.sh
+++ b/dkms/post_remove.sh
@@ -1,6 +1,6 @@
 #!/bin/sh
 
-if [ `find /lib/modules/ -name px4_drv.ko | wc -l` -eq 0 ]; then
+if [ `find /lib/modules/ -name 'px4_drv.ko*' | wc -l` -eq 0 ]; then
     rm -fv /etc/udev/rules.d/90-px4.rules /etc/udev/rules.d/99-px4video.rules
     rm -fv /lib/firmware/it930x-firmware.bin
 fi
diff --git a/driver/Makefile b/driver/Makefile
index 81a76de..90f35ac 100644
--- a/driver/Makefile
+++ b/driver/Makefile
@@ -61,7 +61,7 @@ uninstall:
 	modprobe -r px4_drv; \
 	fi
 	$(cmd_prefix)rm -fv $(INSTALL_DIR)/px4_drv.ko
-	$(cmd_prefix)if [ `find /lib/modules/ -name px4_drv.ko | wc -l` -eq 0 ]; then \
+	$(cmd_prefix)if [ `find /lib/modules/ -name 'px4_drv.ko*' | wc -l` -eq 0 ]; then \
 	rm -fv /etc/udev/rules.d/90-px4.rules /etc/udev/rules.d/99-px4video.rules; \
 	fi
 	$(cmd_prefix)depmod -a $(KVER)