Removed minor bug