Fixes #3 attribute mismatch

Fixes issue #3