o/d problems

Hi all. I have a 1.6l w. an automatic transmission. It has intermittent shifting problems like dropping out of O/D at freeway speeds and shifting hard in/out of gear during a full stop and just having issues shifting in general. A mechanic friend of mine believes that it is the speed sensor...