Fix comment + add cast