Make type hint more generic as it causes drafter to fail