Updated mongoid so after_transition works.