Fixes broken test case