Another small change to setup