Fixed a bug with the last commit. Oops!