Add missing scheduler extension that got dropped in rebase.

Change-Id: I7f82ab569e6cfba82616cb9b01bd8f43e2e0084e