Fix issue 4

This doesn't seem to be used in the moduie?