fixed a small buggy