minor internal fix

Signed-off-by: David Silver <dms.silver@gmail.com>