Removed a line in previous commit. Shouldn't have removed it...