Fixing build.

Small mistake with too many changes that made me miss that tests were
broken