fix another bug in this feature