Fix the getMessage code