Updates wb_s to use new wb_register module