fix BUGS