[Book-DI] Added missing code block