minor code fix for deployment