Minor tweak to level.py