Fix setup imports