Updated main.py again because there was a bug in the original fix

[AadityaAwati]

In the previous version the error handling would only work once, and if the same error occurred again it would just crash like before. So I fixed it, and it's working fine now!