Minor fixes

Signed-off-by: Abhishek Jairath <jairath.abhishek@gmail.com>