fixed a week 2 bug