fix another little bug