small fixes to test section