Fix a small issue with recent changes to this code.