Small change

Please send PR to this repository. @TenninYan