Updated minor bug