Custom module bug fixes