Fix hardcoded code