Fixed code change that was accidentally pulled