Update module.py

Minor fix