maybe we should move it to helper?