Changed test file to adhere to new spec