fix bug in returnStatement