caught some mistakes after my first pull request
caught some mistakes after my first pull request