Fix breaking test