added some missing test code