Fixed build breakage from my previous commit