updated to correct failing test cases
1 file changed