Last state initialized to contain input button
Added a null check (whoops)