Move test into its own test class.