Fixing something

Fix a little bug