Fix bug where update wasn't being called with the right args