Fixes broken test.