fix test for new internal representation