Augmented the tests with basic error validation