move model registration to after_initialize