Fixed a code deficiency