Fixed unittest