diff --git a/neodj.lua b/neodj.lua
index 7712159..2add82b 100644
--- a/neodj.lua
+++ b/neodj.lua
@@ -1,8 +1,8 @@
 accentColor = colors.gray 
 buttonColor = colors.lightGray
 backgroundColor = colors.black
-drive = peripheral.getType("disk_drive")
-if drive == 0 then
+drive = peripheral.find("disk_drive")
+if drive == nil then
 	print "No drive"
 end