fixed a small bug introduced by refactoring