small fix

Signed-off-by: John Hany <johnhany97@gmail.com>