Fix missing code one more time