Update rock_band.py

Small code change.