Fix issue with deploy