Fix missing bug