Forgot to sort something there
Fixing github issue #1