fixed some stuff in the test file