fixed tests for 202 to 204 change on update