tiny changes on test code