Fixing unhandled exceptions