Updated module handling