Add `random` to the import in __init__