Fixed a bug left over from when I first ported this into own repo.