BUG: fix failing test from 682b39e