Actual fix for previous commit