Fix heading collection update issue