This should replace the commented out line (line 120)