Fix forgotten line