Add missing mock functions