Fix code for checking the first argument