test.py: Add error handling. Duh