BLK-228 Updates module initialization and config in the tests

Signed-off-by: Oren Leiman <oleiman@neutral.io>