Changed spec to reflect changes in the code