fixed register.py file