Notification box

Yesterday I posted a fix for this, added an wait for the notification box because the test was sometimes failing. 
But now this is failing because the notification box is not disappearing anymore, so added an assert instead