Fix the assertion error. See #112