--game=Pong argument crashes

[MatPoliquin]

When I run
python3 train_atari.py --game=Pong

It crashes with this callstack:

Traceback (most recent call last):
  File "/usr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap
    self.run()
  File "/usr/lib/python3.6/multiprocessing/process.py", line 93, in run
    self._target(*self._args, **self._kwargs)
  File "/home/fanbingbing/ML/atari-reset/atari_reset/wrappers.py", line 469, in worker
    env = env_fn_wrapper.x()
  File "train_atari.py", line 34, in env_fn
    env = ReplayResetEnv(env, demo_file_name='demos/'+game_name+'.demo', seed=rank, workers_per_sp=workers_per_sp)
  File "/home/fanbingbing/ML/atari-reset/atari_reset/wrappers.py", line 107, in __init__
    assert len(rewards) == len(self.actions)
AssertionError

When I run
python3 train_atari.py --game=MontezumaRevenge

it works fine

Info:
Ubuntu 18.04
Python 3.6.6
synced to latest atari-reset 0c1b112
synced to latest baselines 28aca63