quick fix to reflect changed code