Fixed comment

The comment meant to refer to the import at the top of the file.