Fixed a bug with one of the extension methods.