Fixed minor refactor error from pull request commit
926b9d953dc0e68d12bd8f2ae1ef5fbec26906df