BS-9791
Fix parameter return type