Added fix for dense entity test.