fix breaking test