Amended the code, as per @Shandem's code review.

https://github.com/umbraco/Umbraco-CMS/pull/133