Small code change for order model