fixed a small regression