Fixed the bug with question generation