One more place we needed to update.