made change so it work on new version