Fix test from 61b636c472