Fix the weird bug hopefully