Fix for setup script