Line 2 bug fix