Fixing a test
Fixed a test