fix bug in assisgnment