TST: fixing something but not sure what