fixed bug by upgrade