Minor fix.
fixes #6