Minor fixes after power stage test