#535 fix test