Fix setup method