fixed bug on PR