Minor fix

Signed-off-by: nscavell <nick.scavelli@gmail.com>