fix: add safety