An augmented state feedback controller for a Wheeled Inverted Pendulum (WIP) is proposed in this research. The augmented state feedback controller is able to keep the WIP returning to the origin. Generally, the WIP has both stable and unstable equilibrium points. To keep the WIP over the unstable equilibrium point, the WIP consistently is being controlled. A simple state feedback controller is letting the WIP out of the origin when the center of gravity of the WIP locates out of the schematic center line. In some case of applications, it may not be desirable that the WIP is drifting out of the initial location. The proposed augmented state feedback controller is able to keep the WIP at the initial location whether its center of gravity lies out of the center line or not. Numerical simulations are carried out to show the validation of the augmented sated feedback controller.