Updated code/tests to pass mutant.