Here's a fix

This fixes #1