Fix regression of 487b93956d407b1692b49c51c9e2bae7d36b3b65

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