small fix when importing custom modules