Fix test case

Regression from 4d40bc7