almost fixed a test