unit test now works