Fix broken specs

Not great but it seems to resolve the issue.