Final error fixes