Add Missing code

OMG... Fuck

Signed-off-by: Byon Jae Myong <byon0716@gmail.com>