Corrects minor bugs