Fixed bugs;

Signed-off-by: ashwinh <ashwinh@cybage.com>