fix the learn button