fix bugs
fix baby bug