Small cleanup

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