fix test from last merge