Forgot to change caught class in in test