Fix tests for new specs.