Bug fix

Forgot 'self.'