Minor fix module name

Fix #2