ENH: adjust test to new alogeithm