Add error handling TODO.