fix bad code :)