Updating function to use new parameter

Signed-off-by: Zane M. Kolnik <zanematthew@gmail.com>