Fixed a messing bug