Fix weird bug