Fix wrong parameter.
Fix unit tests.