fix another bug (still buggy)