Fix a little bug, thx Baboon