Actual fix for my bug