Fixing spring code