Renamed Warning into message + trying to correct test

Message seems a better name for the warning. I'm also cleaning the tests