Fix cell shrinking after button hiding