Fixd bug

Fixd bug.