Fix: Push Code

I added this code to fix it when it crashes when selecting push and
without it, it will crash.