Some fixes

Signed-off-by: vaibhav121 <bhasinvaibhav97@gmail.com>