diff --git a/neodj.lua b/neodj.lua
index 82757c9..f259fd6 100644
--- a/neodj.lua
+++ b/neodj.lua
@@ -1,7 +1,10 @@
-local accentColor = colors.gray 
-local buttonColor = colors.lightGray
-local backgroundColor = colors.black
-local drive = peripheral.getType("disk_drive")
+accentColor = colors.gray 
+buttonColor = colors.lightGray
+backgroundColor = colors.black
+drive = peripheral.getType("disk_drive")
+if !drive then
+	print "No drive"
+end
 
 function playDisc()
 	disk.playAudio(drive)