Bug fix (type miss at test cases. oops)