fixing up testing