What toilet part do I need?

My toilet still flushes fine but when you press on the pedal you have to put your toe under it and pull it back up. I'm guessing it's a spring in the water valve assembly -- should I just get the app...