Fix a bug :)