Fix module references