You really need to start thinking about tests, again changes in code and no test cover it