Fix newly added tests
1 file changed