Type error fixes