Import `AssertionFailedException`

Rather than `InvalidArgumentException` which is not used in the code, even if it is the concrete impl.