Small fix to setup