Update line 444 for bug

Updated code to handle a issue