Fix change in setup