Fix possible crash

Signed-off-by: Sup Yut Sum <ch3cooli@gmail.com>