Fixed build (broken in previous commit)