Slightly better unit test