small fix

Signed-off-by: Dave Bush <davebush@gmail.com>