issue #822 fixed

전에 다른 기능을 넣을때 이 부분이 빠진것 같습니다.
버그가 수정되고 잘 동작합니다. 반영해 주세요