From f5997f743ac17dca22e6bc2aa8c1d20eddde77fa Mon Sep 17 00:00:00 2001
From: Roman Simecek <roman@good2go.ch>
Date: Wed, 2 Aug 2017 17:15:22 +0200
Subject: [PATCH] Get the base class so that it works with sti models.

---
 lib/stateful_enum/machine.rb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/stateful_enum/machine.rb b/lib/stateful_enum/machine.rb
index dc2d22a..08890dd 100644
--- a/lib/stateful_enum/machine.rb
+++ b/lib/stateful_enum/machine.rb
@@ -48,7 +48,7 @@ def initialize(model, column, states, prefix, suffix, name, &block)
             if to && (!condition || instance_exec(&condition))
               #TODO transaction?
               instance_eval(&before) if before
-              original_method = self.class.send(:_enum_methods_module).instance_method "#{prefix}#{to}#{suffix}!"
+              original_method = self.class.base_class.send(:_enum_methods_module).instance_method "#{prefix}#{to}#{suffix}!"
               ret = original_method.bind(self).call
               instance_eval(&after) if after
               ret