Updating tests for put/set tests