Add scheduler as a proper type