Bug fix for deploy