Fixed a small mistake