Why was this test removed? Also can you write some tests for the rest of the code please.