[FIX] Correct scheduler execution
[FIX] Add missing import