"Trong sách giáo khoa tiểu học của Việt Nam đọc thấy truyện Cây bút thần của Mã Lương" >> This is as close as possible to your mandarin version, but still sounds natural