Updated code for bug fix