few more fix

few more fix