Fix bug from lint