Updated test for fix