Fixed bug in beam/test/test_beam.py