Fixed a minor problem