minor fix

Signed-off-by: Chris Cheung <chriscsc@me.com>