Fixed a bug (ommitted line of code).