I saw the same episode. It seemed to me that the fish "got the hang of" swimming through the holes. It took them less time as the weeks went on, and they seemed to remember where the holes were.

I found it interesting that the sick fish didn't do well at all, but the fish in the "good tank" learned much more quickly.