tidy up spec code