fix another bug, I think it should work, now