1 files changed, 6 insertions, 1 deletions
diff --git a/src/main/kotlin/org/openrndr/Preload.kt b/src/main/kotlin/org/openrndr/Preload.kt
index 1bbd673..9664416 100644
--- a/src/main/kotlin/org/openrndr/Preload.kt
+++ b/src/main/kotlin/org/openrndr/Preload.kt
@@ -1,14 +1,19 @@
 package org.openrndr
 
 import org.openrndr.draw.*
+import org.openrndr.extensions.Screenshots
+import org.openrndr.extra.olive.*
 import phenax.extensions.HeadlessRecorder
-import phenax.utils.recordingEnabled
+import phenax.extensions.HeadlessScreenshot
+import phenax.utils.*
 
 class Preload : ApplicationPreload() {
   override fun onProgramSetup(program: Program) {
     println("BACKEND(${program.name}): ${program.application::class.qualifiedName}")
     if (recordingEnabled) {
       program.extend(HeadlessRecorder())
+    } else if (screenshotEnabled) {
+      program.extend(HeadlessScreenshot())
     }
   }
 }