Fixed `ValueError when tracking backwards` #111

[soumya997]

Fixed #111, small change, but the if its not done then the demo.ipynb code brakes.

Someone already pointed out the solution, in the issue, but as i couldn't find any PR regarding that so do it myself.

Thank you for your attention.