Fix a small problem