Little fix: changing dangerous code