Fix error messages and TODO