Fixing wrong comparison

See #2