diff --git a/src/ZuluSCSI_main.cpp b/src/ZuluSCSI_main.cpp
index e32ef1fe..5f2d32cf 100644
--- a/src/ZuluSCSI_main.cpp
+++ b/src/ZuluSCSI_main.cpp
@@ -5,10 +5,20 @@
 
 extern "C" int bootloader_main(void);
 
+#ifdef USE_ARDUINO
+extern "C" void setup(void)
+{
+    bootloader_main();
+}
+extern "C" void loop(void)
+{
+}
+#else
 int main(void)
 {
     return bootloader_main();
 }
+#endif
 
 #else
 
diff --git a/src/build_bootloader.py b/src/build_bootloader.py
index 5a97dce2..2f8d757e 100644
--- a/src/build_bootloader.py
+++ b/src/build_bootloader.py
@@ -19,7 +19,7 @@
 for nodelist in env["PIOBUILDFILES"]:
     for node in nodelist:
         filename = str(node.rfile())
-        if 'ZuluSCSI_main.o' not in filename:
+        if 'ZuluSCSI_main' not in filename:
             dep_objs.append(node)
 # print("Bootloader dependencies: ", type(dep_objs), str([str(f.rfile()) for f in dep_objs]))