Fix error after modification #2
Fix error after modification #2