fixes broken test