Fixes test for #233