diff --git a/lib/hotwire-spark.rb b/lib/hotwire-spark.rb
index 79e53a8..015d053 100644
--- a/lib/hotwire-spark.rb
+++ b/lib/hotwire-spark.rb
@@ -24,7 +24,7 @@ def install_into(application)
     end
 
     def enabled?
-      enabled && defined?(Rails::Server)
+      enabled && (defined?(Rails::Server) || $PROGRAM_NAME =~ /puma/)
     end
 
     def cable_server