issue 1, fixed #2

second line of code added