Forgot to commit tests