Add missing self arg