Fix the missing code