Fix integration spec failure introduced in last commit