fixed one bug found another