Fix the assertion error. See #250