Forgot to add class_name parameter return value in last commit