Fix wrong message being printed out
Added a case for when passing integers