Fixed broken Test