Updated tests, but testUpdate is failing. Attempting to fix.