add tests to the fix