NameError: name 'FSMContext' is not defined

Работаю с "Машиной" в aiogram. Выдает ошибку:

@dp.message_handler(state=change.text_key)
async def check_text_key(message: types.Message, state: FSMContext):

Ошибка: NameError: name 'FSMContext' is not defined. В чем проблема?