lib/buildkite/builder/dsl.rb in buildkite-builder-2.0.0.beta1 vs lib/buildkite/builder/dsl.rb in buildkite-builder-2.0.0.beta2

@@ -4,10 +4,14 @@
   module Builder
     class Dsl
       attr_reader :context
 
       def extend(mod)
-        mod < Extension ? super(mod.dsl) : super
+        if mod < Extension
+          super(mod.dsl) if mod.dsl
+        else
+          super
+        end
       end
 
       def initialize(context)
         @context = context
       end