Fixed minnor bug