Fixes #5, thanks @Bashta