Found and fixed an issue with how it was accepting the argument weights.