Yes, that is what I should have said (and it was also what I was thinking at the time, since it is not uncommon for sci-fi episodes to take plots from other stories. https://www.escapistmagazine.com/star-trek-strange-new-world... is another such example).