Forgot to add new tests file