Fix setup bug