Updating Unit Test