fix ugly code...