Fixed 2 small mistakes present in the last pull request.