fixed major issue

see .rb file