Move test into its own section