Fix root is none problem