Fix omnifunc compatibility problem