Minor unittest update