fix for deployment bug

* Quick fix for deployment bug