cleaned out unneeded code from tests