Minor Bug Fix

Minor Bug Fix