Minor setup fix