fix size and add new test method