fix broken UT as a result of new druid api changes