MAINT: fixed bug in test, not sure why I didn't catch that locally