More Fixes

Signed-off-by: GuneetAtwal <harjinder767@gmail.com>