Fixed some minor issues with the test