add fix for task #2