Add missing check on the parameter