Fix missing buttons