Fix unbalanced_source_blocks, they are being allocated now