Minor Bug fixed