Merge uma tests back

re-merge with upstream