Added test, forgot to add in earlier commit