fixed little bug, unstested fix