BUG: fix tests for 2.6.