Fixed a bug form a previous edit