Fix the unittest to reflect the new exception type