Add sanity check on input argument. Oops.