Fixed a 'bug'