Fix test for new completion function