Fix amp bug