Follow up to 6e2b3be

Signed-off-by: ThinkIV <thinkivme@gmail.com>