Updated simple implementation tests for error handling