BUG: Fixed minor bug