Minor fix. Fixes #3