TST: fixed tests failing due to previous commit