Efter hemkomsten från Spanien bosatte sig Staf i Stockholm, där han värvades till den kommunistiska Wollweberligan, vars uppgift var att sabotera fascistisk sjöfart och bygga upp en motståndsrörelse vid en eventuell tysk ockupation av Sverige. Efter att Tyskland angripit Sovjetunionen i juni 1941 arresterades Staf av Säkerhetspolisen tillsammans med större delen av den svenska gruppen, men släpptes i brist på bevis. Han fortsatte då arbeta som sjöman och smugglade illegalt material till ockuperade länder i Europa.