fix breaking tests