Fix variable problem