Comment fix
Same bug fix