Fixed assertion error