update tests for set