Fixed build errors (oops)