Was this already added in #124, and if so, can you produce a clean PR?