Updating unittest