fixing typehints