Updated test to run with code updates