Fix test exception