partition_key discrepancy in rails docs

Rails app migration steps in docs has discrepancy in distribution key that's not present in example rails app.

create_table :page_views, partition_key: :customer_id do |t|
create_distributed_table :page_views, :account_id

I believe that these should be the same value.

Docs link:
http://docs.citusdata.com/en/v8.0/develop/migration_mt_ror.html?highlight=rails#updating-the-rails-application

Rails app examples:
https://github.com/citusdata/citus-example-ad-analytics/blob/master/db/migrate/20160523211642_initial_tables.rb