Added protection :P

Signed-off-by: Kenneth Lim <kenlimmj@gmail.com>