Fixed unit testing!