Fix missing code