Add proper error handling to test