Simple fix for failing test case