Fix missing hunk from previous commit