Button transition fix