diff --git a/automate_farming/database_main.lua b/automate_farming/database_main.lua
index ecbd5c8..a296016 100644
--- a/automate_farming/database_main.lua
+++ b/automate_farming/database_main.lua
@@ -1,2 +1,6 @@
+local databaseStorage = peripheral.find("drive") or error("No drive attached", 0)
+if (databaseStorage.isDiskPresent() == false or databaseStorage.getDiskLabel() ~= "Farming Data") then error("Missing or wrong drive", 0) end
+local databasePath = databaseStorage.getMountPath()
+
 local databaseModem = peripheral.find("modem") or error("No modem attached", 0)
 local databaseStorage = peripheral.find("drive") or error("No modem attached", 0)
\ No newline at end of file