Fix couple broken specs