Small Fix

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