Fixing code so tests run again.