Adding UT to confirm fix.