Fix a small regression