Tests

~ Small fixes in the test script