Modify tests to include chain methods