Fix local variable type