small bug

Fix a small bug