Minor bug fix

Signed-off-by: Prasad Joshi <prasadjoshi.linux@gmail.com>