Did some silly thing, but works. May fix later.