Bug fix

Fix a small bug...