Fixes #124, forgot to include the test