Fixed bug with turning string into class