Fix comment after transfering code to separate module