Made one small change to push notification code