Restore omitted tests