Adding missing self .. doh