Fixing silly mistakes

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