Revert "small fix when importing custom modules"

This reverts commit 571838cc1f8457823e92e858a1f514c6e874faec.