fixed some things in test