better unit testing?