lib/timber/context.rb in timber-1.0.3 vs lib/timber/context.rb in timber-1.0.4

@@ -1,10 +1,16 @@
 module Timber
   # Base class for all `Timber::Contexts::*` classes.
   # @private
   class Context
+    class << self
+      def keyspace
+        @keyspace || raise(NotImplementedError.new)
+      end
+    end
+
     def keyspace
-      raise NoImplementedError.new
+      self.class.keyspace
     end
 
     def as_json(options = {})
       raise NotImplementedError.new
     end
\ No newline at end of file