Updated test
Fix to make test past
Get test to run