Small Fix

Small fixes in unite tests