removing minor bug