fix hard code