Fix #7
Tiny fix for #7