Fixed a minor problem here