Fix for function injection.