adjusted test code