Fix comment

import wrong lib