Fixed a bug ... hopefully.