Fixing test to pass