It can run normally in jupyter 