Forgot to add message back to template