Updating Unit Tests