Put in better unit test for example