Fixed compile error from first pull request. :)