Fix forgotten include