Fixed miss types that caused it not to compile