Fixed a small 'bug'