Update comment to reflect correct file