Fixed build breakage