bug fix

stupid mistake, didnot try if it works