Small fix. Two variables were mistakenly in the wrong place.