Separate test into own function