test fixes & tidy up