Add missing method to mock application