Fix a little bug, sorry