Fix specs

Tests were failing.