Working unit test!