fixed 'bug'