Fixed includes in test directory