fixed typo in beam.py