Fixed rest.py import