Fixed unit test bug. Everything working now.