follow up cleanup

Signed-off-by: Jack Wang <jinpu.wang@profitbricks.com>