MAINT/TST: fixed tests