Forgot to include some tests