Fixed tests for update