Trying to add back tests