Fixes unit invocation.