Fix register model for Mongoid 3.1.0+