Small bug fix

Fixed error with "message" being called