Set STI type column correctly when calling becomes!

The becomes! method is currently broken and need to be called a second time to correctly set the type column when changing from the base model to a sub-class.