Minor fix module names for consistency

Fix #2