Fix #2

Whoops!