Fixing missing check