Fixes broken tests.