better unit tests