and fixed one more little bug