Fix breaking spec