diff --git a/neodj.lua b/neodj.lua
index ed771b0..fc3c041 100644
--- a/neodj.lua
+++ b/neodj.lua
@@ -1,13 +1,14 @@
 local accentColor = colors.gray 
 local buttonColor = colors.lightGray
 local backgroundColor = colors.black
+local drive = peripheral.getType("disk_drive")
 
 function playDisk()
-	disk.playAudio()
+	disk.playAudio(drive)
 end
 
 if disk.isPresent() and disk.hasAudio() then
-	playDisk()
+	playDisk(drive)
 elseif disk.isPresent() then
 	print "Not a music disc"
 else