Fixes so tests run again.