Maybe is good to check the assertion message to make sure it is not other `RuntimeError`