Forgot to add new exception to __init__