[FrameworkBundle] Update the UTs