Fix flying notification on lockscreen after fold

Fixes: 301419977
Test: fold device on lockscreen => no fly
      unfold device on lockscreen => no regression
      swipe away UMO in expanded shade => no regression
      turn on flashlight from lockscreen, change smartspace height
      => no regression

Flag: none
Change-Id: I623d1bb6c6e966ddafef859567d4eecae4e70aa5