Spec tidy up