Fixed conditional in remove method
Merge pull request #8 from andylyall/develop

Fixed conditional in remove method