Managed to confuse myself here. Fixes #2.

Thanks @muan!