Added missing self

Fixes #3