BUG: fixed a bug in the test