fix funny bug