Fix specs

This fixes broken specs.