updated the test with module func implemetation
updated the test with module func implemetation