Fix regression of 12ead18

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