Fixed unit test code

Removed deprecated lines in the configure.ac file and updated the
imports required to make the unit tests work