diff --git a/neodj.lua b/neodj.lua
index f259fd6..d0a05d0 100644
--- a/neodj.lua
+++ b/neodj.lua
@@ -2,7 +2,7 @@ accentColor = colors.gray
 buttonColor = colors.lightGray
 backgroundColor = colors.black
 drive = peripheral.getType("disk_drive")
-if !drive then
+if drive = 0 then
 	print "No drive"
 end