* Update to latest version of AMP optimizer * Update test * Update another test * Update another test