fix for undefined method context_class