Fix a bug. Thanks Fenhl