fix getting module from model map