fix broken test after rebase