Remove TODO for error handling