Minor Bug Fix

See Diff