Override the proper hook to make assign_attributes work with options

This is using a hook added on Rails at
https://github.com/rails/rails/commit/d8b173607da94419e841c4d36f251987c9acf156

Fixes #23