TST: make tests pass