Forgot to add these test classes