fixed both tests to work