Could you do two separate test functions for the two tests?