Fixing test cases after ruby module update