diff --git a/data/dialog-success.sh b/data/dialog-success.sh
index 2abb9c1..a507436 100755
--- a/data/dialog-success.sh
+++ b/data/dialog-success.sh
@@ -1,6 +1,11 @@
 #! /bin/bash
 
-if zenity --question --text ""$1" has been processed successfully. Would you like to reboot for changes to take effect?"
+if echo $1 | grep -i mesa
 then
-	systemctl reboot
+    true
+else
+    if zenity --question --text ""$1" has been processed successfully. Would you like to reboot for changes to take effect?"
+    then
+	    systemctl reboot
+    fi
 fi
diff --git a/data/generate_package_info.sh b/data/generate_package_info.sh
index 942a536..9191c42 100755
--- a/data/generate_package_info.sh
+++ b/data/generate_package_info.sh
@@ -29,7 +29,7 @@ else
 					else
 						if echo "$2" | grep -i -E 'mesa' &> /dev/null
 						then
-							DEVICE="$(lspci | grep -i -E 'vga|display|3d' | cut -d":" -f3)"
+							DEVICE="$(lspci | grep -i -E 'vga|display|3d' | cut -d":" -f3 | grep -vi nvidia)"
 						else
 							DEVICE="$(ubuntu-drivers devices | sed ':a;N;$!ba;s/\nmodel/ /g' | grep vendor | grep -i $2 | sed 's/vendor/Device:/')"
 						fi