fix: added missing check