fix: tests with new error handling