Fix the update test.
Merge pull request #2775 from brendandburns/fix

Fix the update test.