Fix problem with update not working. Test needs to be updated from @jipi
Add test