Contribution Type and Amount not updating on Add Membership form

In version 3.0.0 on the Contact Add Membership form, when the Membership Type is selected, the Contribution Type and Amount automatically update if Record Membership Payment is checked. The onChange javascript call setPaymentBlock has moved in 3.2.2 and now the Contribution Type and Amount are only set when the Record Membership Payment is checked. If Membership Type is selected after checking Record Membership Payment, Contribution Type and Amount are not updated. If Membership Type is changed, Contribution Type and Amount are not updated.

Here is the URL:

http://drupal.demo.civicrm.org/civicrm/contact/view/membership?reset=1&action=add&cid=102&context=membership

Here is the diff of a simple fix:

jbertolacci@imbatmp:~/www.imba.com/sites/default/files/civicrm/custom/php/CRM/Member/Form$ diff Membership.php /var/www/sites/all/modules/civicrm/CRM/Member/Form/Membership.php
366,370c360
< // IMBA – update contribution type mod
< // begin
< //array('onChange' => "buildCustomData( 'Membership', this.value );")
< array('onChange' => "buildCustomData( 'Membership', this.value ); setPaymentBlock( this.value );")
< // end
—
> array('onChange' => "buildCustomData( 'Membership', this.value );")