Fixed test, added update test