fixed a little bug introduced in last commit