updated minor bug